package com.uos.manager.service;


import com.uos.entity.Product;
import com.uos.entity.enums.ProductStatus;
import com.uos.manager.error.ErrorEnum;
import com.uos.manager.repositories.ProductRepository;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;
import org.springframework.util.Assert;

import javax.persistence.criteria.*;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/*
    产品服务类
 */
@Service
public class ProductService {
    private static Logger LOG = LoggerFactory.getLogger(ProductService.class);

    @Autowired
    private ProductRepository repository;
    //添加产品
    public Product addProduct(Product product){
        //打印日志信息
        LOG.debug("创建产品，参数：{}",product);
        //数据校验
        checkProduct(product);
        //设置默认值
        setDefault(product);
        //保存值
        Product result = repository.save(product);
        LOG.debug("创建产品，结果：{}",result);
        return result;
    }
    /*
    设置默认值：
        创建时间、更新时间、投资步长、锁定期
     */
    private void setDefault(Product product) {
        if (product.getCreateAt() == null) {
            product.setCreateAt(new Date());
        }
        if (product.getUpdateAt() == null) {
            product.setUpdateAt(new Date());
        }
        if (product.getStepAmount() == null) {
            product.setStepAmount(BigDecimal.ZERO);
        }
        if (product.getLockTerm() == null) {
            product.setLockTerm(0);
        }
        if (product.getStatus() == null) {
            product.setStatus(ProductStatus.AUDITING.name());
        }
    }

    /*
        产品数据的校验：
        1.非空数据
        2.收益率在0-30%以内
        3.投资步长需要为整数
     */
    private void checkProduct(Product product) {
        Assert.notNull(product.getId(), ErrorEnum.ID_NOT_NULL.getCode());
        //其他非空校验
        Assert.isTrue(BigDecimal.ZERO.compareTo(product.getRewardRate()) < 0 && BigDecimal.valueOf(30).compareTo(product.getRewardRate()) >= 0, "收益率范围错误");
        Assert.isTrue(BigDecimal.valueOf(product.getStepAmount().longValue()).compareTo(product.getStepAmount()) == 0, "投资步长需为整数");
    }
    /**
     * 查询单个产品（简单查询）
     */
    public Product findOne(String id){
        Assert.notNull(id,"需要产品编号参数");
        LOG.debug("查询单个产品,id={}",id);
        Product product = repository.findById(id).orElse(null);
        LOG.debug("查询单个产品，结果={}",product);
        return product;
    }
    /**
     * 分页查询（复杂查询）
     */
    public Page<Product> query(List<String> idList,
                               BigDecimal minRewardRate, BigDecimal maxRewardRate,
                               List<String> statusList,
                               Pageable pageable){
        LOG.debug("查询产品,idList={},minRewardRate={},maxRewardRate={},statusList={},pageable={}",idList,minRewardRate,maxRewardRate,statusList,pageable);

        Specification<Product> specification = new Specification<Product>() {
            @Override
            public Predicate toPredicate(Root<Product> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
                //获取相关的列
                Expression<String> idCol = root.get("id");
                Expression<BigDecimal> rewardRateCol = root.get("rewardRate");
                Expression<String> statusCol = root.get("status");
                //定义断言列表，用来生成断言
                List<Predicate> predicates = new ArrayList<>();
                //判断产品编号的情况
                if (idList != null && idList.size() > 0){
                    predicates.add(idCol.in(idList));
                }
                if (minRewardRate!= null && BigDecimal.ZERO.compareTo(minRewardRate) < 0){
                    predicates.add(cb.ge(rewardRateCol,minRewardRate));
                }
                if (maxRewardRate!= null && BigDecimal.ZERO.compareTo(minRewardRate) < 0){
                    predicates.add(cb.le(rewardRateCol,maxRewardRate));
                }
                //判断产品状态的情况
                if (statusList != null && statusList.size() > 0){
                    predicates.add(statusCol.in(statusList));
                }
                query.where(predicates.toArray(new Predicate[0]));
                return null;
            }
        };
        Page<Product> page = repository.findAll(specification,pageable);
        LOG.debug("查询产品，结果={}",page);
        return page;
    }
}

